Leetcode count items matching a rule python